# coding=utf-8

import openpyxl
import os

'''
编写一个程序，读入几个文本文件的内容（可以自己创造这些文本文件），并
将这些内容插入一个电子表格，每行写入一行文本。第一个文本文件中的行将写入
列 A 中的单元格，第二个文本文件中的行将写入列 B 中的单元格，以此类推
'''


def readTxtToExcel(path):
    dataList = []
    fileList = os.listdir(path)
    for filename in fileList:
        if filename.endswith(".txt"):  # 只读取txt格式文件
            with open(path + filename, "rt") as in_file:
                text = in_file.readlines()
                print(text)  # 观察打印结果，text是一个列表
                dataList.append(text)
        else:
            print("{}\t not txt".format(filename))
    # print(dataList)

    wb = openpyxl.Workbook()
    sheet = wb.active

    for i in range(len(dataList)):
        columnData = dataList[i]
        for j in range(len(columnData)):
            sheet.cell(j + 1, i + 1).value = columnData[j]

    wb.save(path + "openpyxl_test4.xlsx")


if __name__ == "__main__":
    readTxtToExcel("../files/")
